#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date    : 2025-02-01-22:29
# @Author  :
# @File    : export.py
# @Software: PyCharm


# 导入ultralytics的YOLO库
from ultralytics import YOLO
import onnx

"""
导出模型（Exporting）
导出模型是将训练好的模型转换为其他格式，便于部署到不同的平台或设备上
"""

# 加载模型
model = YOLO('runs/model/best.pt')  # 加载自定义的训练模型

# 如果当前脚本作为主程序运行
if __name__ == '__main__':
    model.export(format='onnx',
                 project='runs/onnx'
                 )
